package emailOption;

/**
 * Created by Administrator on 2016/12/7 0002.
 */
import java.io.*;
import java.text.*;
import java.util.*;
import javax.mail.*;
import javax.mail.internet.*;

public class receiveEm {


        private MimeMessage mimeMessage = null;
        private String saveAttachPath = ""; 
        private StringBuffer bodytext = new StringBuffer();
        private String dateformat = "yy-MM-dd HH:mm"; 
    public static int MessageSize;

    public receiveEm(){}
        public receiveEm(MimeMessage mimeMessage) {
            this.mimeMessage = mimeMessage;
        }

        public void setMimeMessage(MimeMessage mimeMessage) {
            this.mimeMessage = mimeMessage;
        }

        public String getFrom() throws Exception {
            InternetAddress address[] = (InternetAddress[]) mimeMessage.getFrom();
            String from = address[0].getAddress();
            if (from == null)
                from = "";
            String personal = address[0].getPersonal();
            if (personal == null)
                personal = "";
            String fromaddr = personal + "<" + from + ">";
            return fromaddr;
        }

        public String getMailAddress(String type) throws Exception {
            String mailaddr = "";
            String addtype = type.toUpperCase();
            InternetAddress[] address = null;
            if (addtype.equals("TO") || addtype.equals("CC")|| addtype.equals("BCC")) {
                if (addtype.equals("TO")) {
                    address = (InternetAddress[]) mimeMessage.getRecipients(Message.RecipientType.TO);
                } else if (addtype.equals("CC")) {
                    address = (InternetAddress[]) mimeMessage.getRecipients(Message.RecipientType.CC);
                } else {
                    address = (InternetAddress[]) mimeMessage.getRecipients(Message.RecipientType.BCC);
                }
                if (address != null) {
                    for (int i = 0; i < address.length; i++) {
                        String email = address[i].getAddress();
                        if (email == null)
                            email = "";
                        else {
                            email = MimeUtility.decodeText(email);
                        }
                        String personal = address[i].getPersonal();
                        if (personal == null)
                            personal = "";
                        else {
                            personal = MimeUtility.decodeText(personal);
                        }
                        String compositeto = personal + "<" + email + ">";
                        mailaddr += "," + compositeto;
                    }
                    mailaddr = mailaddr.substring(1);
                }
            } else {
                throw new Exception("Error emailaddr type!");
            }
            return mailaddr;
        }


        public String getSubject() throws MessagingException {
            String subject = "";
            try {
                subject = MimeUtility.decodeText(mimeMessage.getSubject());
                if (subject == null)
                    subject = "";
            } catch (Exception exce) {}
            return subject;
        }

        public String getSentDate() throws Exception {
            Date sentdate = mimeMessage.getSentDate();
            SimpleDateFormat format = new SimpleDateFormat(dateformat);
            return format.format(sentdate);
        }

  
        public String getBodyText() {
            return bodytext.toString();
        }

 
        public void getMailContent(Part part) throws Exception {
            String contenttype = part.getContentType();
            int nameindex = contenttype.indexOf("name");
            boolean conname = false;
            if (nameindex != -1)
                conname = true;
            System.out.println("CONTENTTYPE: " + contenttype);
            if (part.isMimeType("text/plain") && !conname) {
                bodytext.append((String) part.getContent());
            } else if (part.isMimeType("text/html") && !conname) {
                bodytext.append((String) part.getContent());
            } else if (part.isMimeType("multipart/*")) {
                Multipart multipart = (Multipart) part.getContent();
                int counts = multipart.getCount();
                for (int i = 0; i < counts; i++) {
                    getMailContent(multipart.getBodyPart(i));
                }
            } else if (part.isMimeType("message/rfc822")) {
                getMailContent((Part) part.getContent());
            } else {}
        }


        public boolean getReplySign() throws MessagingException {
            boolean replysign = false;
            String needreply[] = mimeMessage
                    .getHeader("Disposition-Notification-To");
            if (needreply != null) {
                replysign = true;
            }
            return replysign;
        }


        public String getMessageId() throws MessagingException {
            return mimeMessage.getMessageID();
        }

 
        public boolean isNew() throws MessagingException {
            boolean isnew = false;
            Flags flags = ((Message) mimeMessage).getFlags();
            Flags.Flag[] flag = flags.getSystemFlags();
            System.out.println("flags's length: " + flag.length);
            for (int i = 0; i < flag.length; i++) {
                if (flag[i] == Flags.Flag.SEEN) {
                    isnew = true;
                    System.out.println("seen Message.......");
                    break;
                }
            }
            return isnew;
        }

 
        public boolean isContainAttach(Part part) throws Exception {
            boolean attachflag = false;
            String contentType = part.getContentType();
            if (part.isMimeType("multipart/*")) {
                Multipart mp = (Multipart) part.getContent();
                for (int i = 0; i < mp.getCount(); i++) {
                    BodyPart mpart = mp.getBodyPart(i);
                    String disposition = mpart.getDisposition();
                    if ((disposition != null)
                            && ((disposition.equals(Part.ATTACHMENT)) || (disposition
                            .equals(Part.INLINE))))
                        attachflag = true;
                    else if (mpart.isMimeType("multipart/*")) {
                        attachflag = isContainAttach((Part) mpart);
                    } else {
                        String contype = mpart.getContentType();
                        if (contype.toLowerCase().indexOf("application") != -1)
                            attachflag = true;
                        if (contype.toLowerCase().indexOf("name") != -1)
                            attachflag = true;
                    }
                }
            } else if (part.isMimeType("message/rfc822")) {
                attachflag = isContainAttach((Part) part.getContent());
            }
            return attachflag;
        }

 
        public void saveAttachMent(Part part) throws Exception {
            String fileName = "";
            if (part.isMimeType("multipart/*")) {
                Multipart mp = (Multipart) part.getContent();
                for (int i = 0; i < mp.getCount(); i++) {
                    BodyPart mpart = mp.getBodyPart(i);
                    String disposition = mpart.getDisposition();
                    if ((disposition != null)
                            && ((disposition.equals(Part.ATTACHMENT)) || (disposition
                            .equals(Part.INLINE)))) {
                        fileName = mpart.getFileName();
                        if (fileName.toLowerCase().indexOf("gb2312") != -1) {
                            fileName = MimeUtility.decodeText(fileName);
                        }
                        saveFile(fileName, mpart.getInputStream());
                    } else if (mpart.isMimeType("multipart/*")) {
                        saveAttachMent(mpart);
                    } else {
                        fileName = mpart.getFileName();
                        if ((fileName != null)
                                && (fileName.toLowerCase().indexOf("GB2312") != -1)) {
                            fileName = MimeUtility.decodeText(fileName);
                            saveFile(fileName, mpart.getInputStream());
                        }
                    }
                }
            } else if (part.isMimeType("message/rfc822")) {
                saveAttachMent((Part) part.getContent());
            }
        }



        public void setAttachPath(String attachpath) {
            this.saveAttachPath = attachpath;
        }


        public void setDateFormat(String format) throws Exception {
            this.dateformat = format;
        }

 
        public String getAttachPath() {
            return saveAttachPath;
        }

 
        private void saveFile(String fileName, InputStream in) throws Exception {
            String osName = System.getProperty("os.name");
            String storedir = getAttachPath();
            String separator = "";
            if (osName == null)
                osName = "";
            if (osName.toLowerCase().indexOf("win") != -1) {
                separator = "\\";
                if (storedir == null || storedir.equals(""))
                    storedir = "c:\\tmp";
            } else {
                separator = "/";
                storedir = "/tmp";
            }
            File storefile = new File(storedir + separator + fileName);
            System.out.println("storefile's path: " + storefile.toString());
            // for(int i=0;storefile.exists();i++){
            // storefile = new File(storedir+separator+fileName+i);
            // }
            BufferedOutputStream bos = null;
            BufferedInputStream bis = null;
            try {
                bos = new BufferedOutputStream(new FileOutputStream(storefile));
                bis = new BufferedInputStream(in);
                int c;
                while ((c = bis.read()) != -1) {
                    bos.write(c);
                    bos.flush();
                }
            } catch (Exception exception) {
                exception.printStackTrace();
                throw new Exception("文件保存失败!");
            } finally {
                bos.close();
                bis.close();
            }
        }


        public static ArrayList<receiveEm> Receive(String user,String password) throws Exception {
            Properties props = System.getProperties();
            ArrayList<receiveEm> content=new ArrayList<>();
            props.put("mail.smtp.host", "smtp.sina.com.cn");
            props.put("mail.smtp.auth", "true");
            Session session = Session.getDefaultInstance(props, null);
            URLName urln = new URLName("pop3", "pop3.sina.com.cn", 110, null,
                    user, password);
            Store store = session.getStore(urln);
            store.connect();
            Folder folder = store.getFolder("INBOX");
            folder.open(Folder.READ_ONLY);
            Message message[] = folder.getMessages();
            MessageSize=message.length;
            System.out.println("Messages's length: " + message.length);
            receiveEm pmm = null;
            for (int i = 0; i < message.length; i++) {
                System.out.println("======================");
                pmm = new receiveEm((MimeMessage) message[i]);
                System.out.println("Message " + i + " subject: " + pmm.getSubject());
                System.out.println("Message " + i + " sentdate: "+ pmm.getSentDate());
                System.out.println("Message " + i + " replysign: "+ pmm.getReplySign());
                System.out.println("Message " + i + " hasRead: " + pmm.isNew());
                System.out.println("Message " + i + "  containAttachment: "+ pmm.isContainAttach((Part) message[i]));
                System.out.println("Message " + i + " form: " + pmm.getFrom());
                System.out.println("Message " + i + " to: "+ pmm.getMailAddress("to"));
                System.out.println("Message " + i + " cc: "+ pmm.getMailAddress("cc"));
                System.out.println("Message " + i + " bcc: "+ pmm.getMailAddress("bcc"));
                System.out.println("Message " + i + " sentdate: "+ pmm.getSentDate());
                System.out.println("Message " + i + " Message-ID: "+ pmm.getMessageId());
                pmm.getMailContent((Part) message[i]);
                System.out.println("Message " + i + " bodycontent: \r\n"
                        + pmm.getBodyText());
                System.out.println("======================");
                pmm.setAttachPath("D:\\");
                pmm.saveAttachMent((Part) message[i]);
                content.add(pmm);
            }
            return content;
        }
    }
